REPO_DIR=/home/boai/APO
NUM_GPUS=1
MICRO_BATCH_SIZE=16
SOURCE_FILE=/home/boai/APO/data/hh-split/rm_data/hh_split_rm_alpaca_v0.sample.json
REFERENCE_FILE=/home/boai/APO/data/hh-split/rm_data/hh_split_rm.train.json
GOLD_FILE=/home/boai/APO/data/hh-split/rm_data/hh_split_rm.golden.json
MODEL_PATH=/home/boai/LLM_models/llama-2-7b-hf/
OUTPUT_PATH=/home/boai/APO/data/hh-split/tlpo_data/rm_tlpo_data_v1_text_scores.json

torchrun --nproc_per_node=${NUM_GPUS} --master_port=6000 ${REPO_DIR}/generate_pn_dataset.py \
    --source_fil ${SOURCE_FILE} \
    --reference_file ${REFERENCE_FILE} \
    --gold_file ${GOLD_FILE} \
    --model_path ${MODEL_PATH} \
    --output_path ${OUTPUT_PATH}
